#include "wifi_network.h"

const byte DNS_PORT = 53;                         //设置DNS端口号
const int webPort = 80;                           //设置Web端口号
const int resetPin = 0;                           //设置重置按键引脚,用于删除WiFi信息
const int LED = 48;                               //设置LED引脚

String scanNetworksID = "";                       //用于储存扫描到的WiFi ID
int connectTimeOut_s = 15;                        //WiFi连接超时时间，单位秒

String wifi_ssid = "";                            //暂时存储wifi账号密码
String wifi_pass = "";                            //暂时存储wifi账号密码

IPAddress apIP(192, 168, 4, 1);                   //设置AP的IP地址
DNSServer dnsServer;                              //创建dnsServer实例
WebServer server(webPort);                        //开启web服务, 创建TCP SERVER,参数: 端口号,最大连接数 


 //void connectToWiFi(int timeOut_s);

Network::Network(char* ap_ssid, char* host_name)
{
  this->AP_SSID = ap_ssid;
  this->HOST_NAME = host_name;
  WiFi.hostname(this->HOST_NAME);             //设置设备名
}

//初始化AP模式
void Network::initSoftAP()
{
  WiFi.mode(WIFI_AP);     //配置为AP模式
  WiFi.softAPConfig(apIP, apIP, IPAddress(255, 255, 255, 0));   //设置AP热点IP和子网掩码
  if(WiFi.softAP(AP_SSID)){   //开启AP热点,如需要密码则添加第二个参数
    //打印相关信息
    Serial.println("ESP32S3-PixelClock SoftAP is right.");
    Serial.print("Soft-AP IP address = ");
    Serial.println(WiFi.softAPIP());
    Serial.println(String("MAC address = ")  + WiFi.softAPmacAddress().c_str());
  }else{  //开启热点失败
    Serial.println("WiFiAP Failed");
    delay(1000);
    Serial.println("restart now...");
    ESP.restart();  //重启复位esp32
  }
}
 
//初始化DNS服务器
void Network::initDNS()
{ 
  //判断将所有地址映射到esp32的ip上是否成功
  if(dnsServer.start(DNS_PORT, "*", apIP)){ 
    Serial.println("start dnsserver success.");
  }else{
    Serial.println("start dnsserver failed.");
  }
}
//用于配置WiFi
void Network::wifiConfig()
{
  initSoftAP();
  initDNS();
  initWebServer();
  scanWiFi();
}
void Network::connectToWiFi(int timeOut_s)
{
      Serial.println("进入connectToWiFi()函数");
      //设置为STA模式并连接WIFI
      WiFi.mode(WIFI_STA);
      //WiFi.setAutoConnect(true);//设置自动连接
      //用字符串成员函数c_str()生成一个const char*指针，指向以空字符终止的数组,即获取该字符串的指针。
      if(wifi_ssid !=""){
          Serial.println("用web配置信息连接.");
          
          WiFi.begin(wifi_ssid.c_str(), wifi_pass.c_str());
          wifi_ssid = "";
          wifi_pass = "";
        }
        // else{
        //    Serial.println("用nvs保存的信息连接.");
        //    WiFi.begin();//连接上一次连接成功的wifi
        // }

      int Connect_time = 0; //用于连接计时，如果长时间连接不成功，复位设备
      while (WiFi.status() != WL_CONNECTED) {  //等待WIFI连接成功
        Serial.print(".");
        digitalWrite(LED,!digitalRead(LED));
        delay(500);
        Connect_time ++;
        if (Connect_time > 2*timeOut_s) {  //长时间连接不上，重新进入配网页面
          digitalWrite(LED,LOW);
          Serial.println("");
          Serial.println("WIFI autoconnect fail, start AP for webconfig now...");
          wifiConfig();   //转到网页端手动配置wifi
          return;         //跳出 防止无限初始化
          //break;        //跳出 防止无限初始化
        }
      }
      if(WiFi.status() == WL_CONNECTED){
          Serial.println("WIFI connect Success");
          Serial.printf("SSID:%s", WiFi.SSID().c_str());
          Serial.printf(", PSW:%s\r\n", WiFi.psk().c_str());
          Serial.print("LocalIP:");
          Serial.print(WiFi.localIP());
          Serial.print(" ,GateIP:");
          Serial.println(WiFi.gatewayIP());
          Serial.print("WIFI status is:");
          Serial.print(WiFi.status());
          digitalWrite(LED,HIGH);      
          server.stop();
      }
}
 

//处理网站根目录“/”(首页)的访问请求,将显示配置wifi的HTML页面
void handleRoot(){
  if (server.hasArg("selectSSID")){
      server.send(200, "text/html", ROOT_HTML + scanNetworksID + "</body></html>");
    }else{
      server.send(200, "text/html", ROOT_HTML + scanNetworksID + "</body></html>");
    }     
}
 
void handleConfigWifi()
{
  Network network("ESP32S3-PixelClock","ESP32S3-PixelClock");//AP热点名称,设置设备名

      //返回http状态
      //server.send(200, "text/html", SUCCESS_HTML);
      if (server.hasArg("ssid")) {//判断是否有账号参数
        Serial.print("got ssid:");
        wifi_ssid = server.arg("ssid");      //获取html表单输入框name名为"ssid"的内容
        // strcpy(sta_ssid, server.arg("ssid").c_str());//将账号参数拷贝到sta_ssid中
        Serial.println(wifi_ssid);
      }else{//没有参数
        Serial.println("error, not found ssid");
        server.send(200, "text/html", "<meta charset='UTF-8'>error, not found ssid");//返回错误页面
        return;
      }
      //密码与账号同理
      if (server.hasArg("pass")) {
        Serial.print("got password:");
        wifi_pass = server.arg("pass");       //获取html表单输入框name名为"pwd"的内容
        //strcpy(sta_pass, server.arg("pass").c_str());
        Serial.println(wifi_pass);
      }else{
        Serial.println("error, not found password");
        server.send(200, "text/html", "<meta charset='UTF-8'>error, not found password");
        return;
      }
      server.send(200, "text/html", "<meta charset='UTF-8'>SSID："+wifi_ssid+"<br />password:"+wifi_pass+"<br />已取得WiFi信息,正在尝试连接,请手动关闭此页面。");//返回保存成功页面      
      delay(2000);    
      WiFi.softAPdisconnect(true);     //参数设置为true，设备将直接关闭接入点模式，即关闭设备所建立的WiFi网络。 
      server.close();                  //关闭web服务       
      WiFi.softAPdisconnect();         //在不输入参数的情况下调用该函数,将关闭接入点模式,并将当前配置的AP热点网络名和密码设置为空值.    
      Serial.println("WiFi Connect SSID:" + wifi_ssid + "  PASS:" + wifi_pass);   
      if(WiFi.status() != WL_CONNECTED){
        Serial.println("开始调用连接函数connectToWiFi()..");
        network.connectToWiFi(connectTimeOut_s);
      }else{
        Serial.println("提交的配置信息自动连接成功..");
      }    
}

// 设置处理404情况的函数'handleNotFound'
void handleNotFound(){            // 当浏览器请求的网络资源无法在服务器找到时通过此自定义函数处理
     handleRoot();                 //访问不存在目录则返回配置页面
//   server.send(404, "text/plain", "404: Not found");   
}
//初始化WebServer
void Network::initWebServer(){
  //给设备设定域名esp32,完整的域名是esp32.local ??
  if(MDNS.begin("esp32")){
    Serial.println("MDNS responder started");
  }
 //必须添加第二个参数HTTP_GET，以下面这种格式去写，否则无法强制门户
  server.on("/", HTTP_GET, handleRoot);                      //  当浏览器请求服务器根目录(网站首页)时调用自定义函数handleRoot处理，设置主页回调函数，必须添加第二个参数HTTP_GET，否则无法强制门户
  server.on("/configwifi", HTTP_POST, handleConfigWifi);     //  当浏览器请求服务器/configwifi(表单字段)目录时调用自定义函数handleConfigWifi处理
  server.onNotFound(handleNotFound);                         //当浏览器请求的网络资源无法在服务器找到时调用自定义函数handleNotFound处理 
  //Tells the server to begin listening for incoming connections.Returns None
  server.begin();                                           //启动TCP SERVER
//server.setNoDelay(true);                                  //关闭延时发送
  Serial.println("WebServer started!");
}

//扫描WiFi
bool Network::scanWiFi(){
    Serial.println("scan start");
  // 扫描附近WiFi
  int n = WiFi.scanNetworks();
  Serial.println("scan done");
  if (n == 0) {
    Serial.println("no networks found");
    scanNetworksID = "no networks found";
    return false;
  }else{
    Serial.print(n);
    Serial.println(" networks found");
    for (int i = 0; i < n; ++i) {
      // Print SSID and RSSI for each network found
      Serial.print(i + 1);
      Serial.print(": ");
      Serial.print(WiFi.SSID(i));
      Serial.print(" (");
      Serial.print(WiFi.RSSI(i));
      Serial.print(")");
      Serial.println((WiFi.encryptionType(i) == WIFI_AUTH_OPEN) ? " " : "*");
      scanNetworksID += "<P>" + WiFi.SSID(i) + "</P>";
      delay(10);
    }
    return true;
  }
}
 


//LED闪烁,led为脚号,n为次数,t为时间间隔ms
void Network::blinkLED(int led,int n,int t){
  for(int i=0;i<2*n;i++){
     digitalWrite(led,!digitalRead(led));
     delay(t);
   }   
 }

//删除保存的wifi信息,并使LED闪烁5次
void Network::restoreWiFi(){
       delay(500);
       esp_wifi_restore();  //删除保存的wifi信息
       Serial.println("连接信息已清空,准备重启设备..");
       delay(10);           
       blinkLED(LED,5,500); //LED闪烁5次
       digitalWrite(LED,LOW);
  }

int Network::checkConnect(bool reConnect)
{
    if(WiFi.status() != WL_CONNECTED)
    {
        if(reConnect == true && WiFi.getMode() != WIFI_AP && WiFi.getMode() != WIFI_AP_STA )
        {
            Serial.println("WIFI未连接.");
            Serial.println("WiFi Mode:");
            Serial.println(WiFi.getMode());
            Serial.println("正在连接WiFi...");
            this->connectToWiFi(connectTimeOut_s);
            Serial.println("WiFi Mode:");
            Serial.println(WiFi.getMode());
            if(WiFi.status() == WL_CONNECTED && WiFi.getMode() == WIFI_STA )return 1;
        }  
    }else if(WiFi.status() == WL_CONNECTED && WiFi.getMode() == WIFI_STA )//2=WIFI_STA
    {
      return 1;
    }
    return 0;
  }

int Network::networkTask(void)
{
    dnsServer.processNextRequest();   //检查客户端DNS请求
    server.handleClient();            //检查客户端(浏览器)http请求
    return checkConnect(true);               //检测网络连接状态，参数true表示如果断开重新连接
}


// void setup() {
//   pinMode(LED,OUTPUT);                  //配置LED口为输出口
//   digitalWrite(LED,LOW);                //初始灯灭
//   pinMode(resetPin, INPUT_PULLUP);      //按键上拉输入模式(默认高电平输入,按下时下拉接到低电平)
//   Serial.begin(baudRate);
//   WiFi.hostname(HOST_NAME);             //设置设备名
 
//   connectToWiFi(connectTimeOut_s);
// }
 
// void loop() {
//       //长按5秒(P0)清除网络配置信息
//     if(digitalRead(resetPin)){
//         delay(5000);
//         if(!digitalRead(resetPin)){   
//            Serial.println("\n按键已长按5秒,正在清空网络连保存接信息.");  
//            restoreWiFi();    //删除保存的wifi信息 
//            ESP.restart();    //重启复位esp32
//            Serial.println("已重启设备.");
//         }      
//      }
 
//     dnsServer.processNextRequest();   //检查客户端DNS请求
//     server.handleClient();            //检查客户端(浏览器)http请求
//     checkConnect(true);               //检测网络连接状态，参数true表示如果断开重新连接
    
//     delay(30);
// }